Die Speccy' 97/Die Speccy' 97.iso/amiga_system/the_aminet/util/wb/peelnewtools.lha/PeelNewTools.c

C/C++ Source or Header | 1995-09-09 | 4KB | 204 lines

#define ARRAYSIZE 1000

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

#include <exec/types.h>
#include <exec/memory.h>
#include <libraries/dos.h>
#include <clib/exec_protos.h>
#include <clib/dos_protos.h>
#include <clib/icon_protos.h>
#include <dos/rdargs.h>

char version[]="VER: PeelNewTools V1.0▀";
char filename[40];
char longdirname[160]="";
char template[]="File/A,S=Single/S,R=Recursive/S,L=Log/S";
LONG argarray[4];
char **newtools;
struct RDArgs *rdargs;

BOOL infoname(char *name)
{
    int length=strlen(name);

    if(strlen(name)<5)
        return(FALSE);
    if(stricmp(name+length-5,".info"))
        return(FALSE);
    return(TRUE);
}

void pferror( void )
{
        PrintFault(IoErr(),"ERROR: ");
}

void safeexit(int x)
{
    if(newtools)
        FreeMem(newtools,ARRAYSIZE);

    FreeArgs(rdargs);
    exit(x);
}

int cleantools(char *filename)
{
    struct DiskObject *icon;
    char **tools1,**tools2,**oldtools;
    int error=0;

    if(!(icon=GetDiskObject(filename))) {
        pferror();
        return(5);
    }

    if(tools1=oldtools=icon->do_ToolTypes) {
        tools2=newtools;

        while(*tools1)
            if(stpblk(*tools1)[0]==0)
                tools1++;
            else
                if(!(strncmp(*tools1,"IM1=",4))) {
                    tools1++;
                }
                else
                    if(!(strncmp(*tools1,"IM2=",4))) {
                        tools1++;
                    }
                    else
                        if(!(strcmp(*tools1,"*** DON'T EDIT THE FOLLOWING LINES!! ***"))) {
                            tools1++;
                        }
                        else
                            *(tools2++)=*(tools1++);

        *tools2=NULL;

        icon->do_ToolTypes=newtools;
        if(!(PutDiskObject(filename,icon))) {
            pferror();
            error=5;
        }

        icon->do_ToolTypes=oldtools;
    }

    FreeDiskObject(icon);
    return(error);
}

void exitscan(BPTR oldlock,BPTR dirlock,struct FileInfoBlock *fblock)
{
    if(oldlock)
        CurrentDir(oldlock);
    FreeMem(fblock,sizeof(struct FileInfoBlock));
    UnLock(dirlock);
}

int scandir(char *dirname)
{
    BPTR    dirlock,oldlock=NULL;
    struct FileInfoBlock *fiblock=NULL;
    char *fname;
    int error=0,errror;
    int longlength;

    if(!(dirlock=Lock(dirname,ACCESS_READ))) {
        pferror();
        return(5);
    }

    if(!(fiblock = AllocMem( sizeof( struct FileInfoBlock ),
                                    MEMF_PUBLIC | MEMF_CLEAR ))) {
        UnLock(dirlock);
        PutStr("ERROR: Out of memory.\n");
        return(20);
    }

    if(!(Examine(dirlock,fiblock))) {
        pferror();
        exitscan(NULL,dirlock,fiblock);
        return(5);
    }

    if(fiblock->fib_DirEntryType<=0) {
        printf("ERROR: %s is not a directory.\n",dirname);
        exitscan(NULL,dirlock,fiblock);
        return(10);
    }

    strcat(longdirname,dirname);
    longlength=strlen(longdirname);
    if(longdirname[longlength-1]!='/') {
        longdirname[longlength++]='/';
        longdirname[longlength]='\0';
    }

    oldlock=CurrentDir(dirlock);

    while(ExNext(dirlock,fiblock)) {
        fname=fiblock->fib_FileName;
        if((fiblock->fib_DirEntryType>0)&&argarray[2]) {
            errror=scandir(fname);
            if(errror==20) {
                PutStr("CleanInfo aborted..\n");
                exitscan(oldlock,dirlock,fiblock);
                return(20);
            }
            error=(errror>error)? errror : error;
        }
        else
            if(infoname(fname)) {
                if(argarray[3])
                    printf("%s%s...\n",longdirname,fname);
                fname[strlen(fname)-5]=0;
                error=(cleantools(fname)>error)? 5 : error;
            }
    }
    longdirname[strlen(longdirname)-strlen(dirname)]=0;
    exitscan(oldlock,dirlock,fiblock);
    return(error);
}

int main(int argc, char *argv[])
{
    int error=0;

    if(!(newtools=(char **)AllocMem(ARRAYSIZE,MEMF_ANY))) {
        PutStr("ERROR: Out of memory.\n");
        safeexit(20);
    }

    if(!(rdargs=ReadArgs(template,argarray,NULL)))
        safeexit(5);

    strncpy(filename,(char *)argarray[0],39);

    if(argarray[1]) {
        if(infoname(filename))
            filename[strlen(filename)-5]=0;
        error=cleantools(filename);
        if(argarray[3])
            printf("\nCleaning %s.info\n\n",filename);
    }
    else {
        if(argarray[3])
            PutStr("\nCleaning files:\n");
        error=scandir(filename);
        if(argarray[3]) {
            PutStr("\nDone..\n\n");
            if(error)
                PutStr("Operation not entirely successful.\n\n");
            else
                PutStr("Operation successful.\n\n");
        }
    }

    safeexit(error);
}